# -*- coding: utf-8 -*-
from django.contrib import admin
from models import Empresa, Publicidade, Local
class EmpresaAdmin(admin.ModelAdmin):
    fieldsets = [
        ('Principais', {'fields':('nome', 'cnpj', ('telefone', 'website'))}),
        ('Logradouro', {'fields':('endereco',('bairro', 'cep'), ('cidade', 'uf'))}),
        ('Responsável para contato', {'fields':(('responsavel', 'telefoneresponsavel'),)}),
        ('Extras', {'fields':['logo']}),
    ]
    list_display = ['nome', 'responsavel', 'telefone', 'website']
    search_fields = ['nome']
    save_on_top = True
    ordering = ['nome']

class PublicidadeAdmin(admin.ModelAdmin):
    fieldsets = [
        ('Principais', {'fields':('ativo', 'nome')}),
        ('Datas', {'fields':(('dataentrada', 'datasaida'), 'nomeEmpresa')}),
        ('Publicidade', {'fields':('tipo', 'link', 'arquivo')}),
    ]
    list_display = ['nome', 'link', 'tipo', 'dataentrada', 'datasaida', 'ativo', 'foto_publicidade']
    search_fields = ['nome', 'nomeEmpresa', 'link']
    list_filter = ['tipo']
    ordering = ['-dataentrada']
    def foto_publicidade(self, obj):
        if obj.arquivo:
            return u'<img src=/media/%s>' % (obj.arquivo)
        else:
            return u'<img src=/media/img/avatar.png>'

    foto_publicidade.short_description = 'Fotografia'
    foto_publicidade.allow_tags = True

class LocalAdmin(admin.ModelAdmin):
    fieldsets = [
        ('Empresa', {'fields':['nomeEmpresa']}),
        ('Principais', {'fields':(('publicar', 'manchete'),('titulo'), ('conteudo'),)}),
        ('Mídia', {'fields':['imagem']}),
    ]

admin.site.register(Empresa, EmpresaAdmin)
admin.site.register(Publicidade, PublicidadeAdmin)
admin.site.register(Local, LocalAdmin)
